import torch
from torch import nn


if __name__ == '__main__':
    nn.MultiheadAttention()
    
    